package com.hbase.utils;

/**
 * @ClassName DataSourceContextHolder
 * @Description TODO
 * @Author zhurs
 * @Date 2019/10/28  9:39
 * @Version 1.0
 **/
public class DataSourceContextHolder {

    private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();

    //设置要使用的数据源
    public static void setDbType(String dbType) {
        contextHolder.set(dbType);
    }

    //获取数据源
    public static String getDbType() {
        return contextHolder.get();
    }

    //清除数据源，使用默认的数据源
    public static void clearDbType() {
        contextHolder.remove();
    }

}
